#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>

#define NUM 1024//表示可以输入的命令行字符个数
#define SIZE 64
#define SEP " "
//#define Debug 1

char cwd[1024];//用于存放当前工作路径（每次输入指令前面的提示信息是要用到的），不能定义在函数里面，不然函数调用结束就被销毁
char enval[1024]; // for test
int lastcode = 0;//最近一个进程退出对应的退出码

char *homepath()//只输入cd时，会进入家目录，而这个路径刚好保存在一个环境变量HONE中
{
    char *home = getenv("HOME");
    if(home) return home;
    else return (char*)".";
}

const char *getUsername()//获取用户名字
{
    const char *name = getenv("USER");
    if(name) return name;
    else return "none";
}
const char *getHostname()//获取主机名
{
    const char *hostname = getenv("HOSTNAME");
    if(hostname) return hostname;
    else return "none";
}
const char *getCwd()//获取当前工作目录
{
    const char *cwd = getenv("PWD");
    if(cwd) return cwd;
    else return "none";
}
int getUserCommand(char *command, int num)
{
    printf("[%s@%s %s]# ", getUsername(), getHostname(), getCwd());//就是模拟输入指令前面提示的语句（这里不加换行符就能在同一行输入）
    char *r = fgets(command, num, stdin); // 最终还是得输入\n（scanf遇到空格会停止，而很多指令会有空格）
    if(r == NULL) return -1;
    //比如输入"abcd\n"，如果不要"\n"，可以把它改成斜杠零（因为不想输入指令之后自动换行）
    command[strlen(command) - 1] = '\0'; // 不可能越界，因为就算什么都不输入也得按回车，也就是说最少都有一个"\n"
    return strlen(command);
}

void commandSplit(char *in, char *out[])
{
    int argc = 0;
    out[argc++] = strtok(in, SEP);
    while( out[argc++] = strtok(NULL, SEP));

#ifdef Debug
    for(int i = 0; out[i]; i++)//输出每次输入的指令分割之后的各段内容，用于测试是否能够正常分割
    {
        printf("%d:%s\n", i, out[i]);
    }
#endif
}

int execute(char *argv[])
{
    pid_t id = fork();//注意要创建子进程再进行程序替换，直接在父进程程序替换会导致后面的代码都被换掉
    if(id < 0) return -1;//表示空串不用怎么处理
    else if(id == 0) //child
    {
        // exec command
        execvp(argv[0], argv); // cd ..这种内建命令无法在子进程执行，因为执行的是子进程，对父进程没有影响
        exit(1);
    }
    else // father
    {
        int status = 0;
        pid_t rid = waitpid(id, &status, 0);
        if(rid > 0){
            lastcode = WEXITSTATUS(status);//获取进程退出码
        }
    }

    return 0;
}

void cd(const char *path)//注意要修改环境变量中的pwd内容，因为cd之后的路径如果改变了，那么输入的指令前面的提示词也应该改变
//而这部分内容是通过环境变量相关的函数获取的，所以要及时修改
{
    chdir(path);
    char tmp[1024];
    getcwd(tmp, sizeof(tmp));
    sprintf(cwd, "PWD=%s", tmp); // bug
    putenv(cwd);//将环境变量的pwd内容修改为最新的内容（覆盖pwd的内容），其他内容不变
}

// 内建命令就是bash自己执行的，类似于自己内部的一个函数！
// 1->yes, 0->no, -1->err
int doBuildin(char *argv[])//内建命令需要由父进程也就是mybash自己来执行
{
    if(strcmp(argv[0], "cd") == 0)//不能在子进程中执行，不然就是只有子进程在切换目录，父进程仍然在当前目录
    {
        char *path = NULL;
        if(argv[1] == NULL) path=homepath();
        else path = argv[1];
        cd(path);
        return 1;
    }
    else if(strcmp(argv[0], "export") == 0)
    {
        if(argv[1] == NULL) return 1;
        strcpy(enval, argv[1]);
        putenv(enval); 
        //putenv(argv[1]);//只写这个语句本来也可以添加环境变量，但是下一次查看时添加的环境变量就不见了，因为argv数组指向的内容来自
        //usercommand，而在main函数中每次循环都会刷新这两个变量里的缓冲区，可能下一次查看的时候就找不到了
        //因此定义了一个全局变量enval（简易版，用来测试），一般用二维数组或者字符指针数组更好，因为能放的更多而且不会被覆盖
        return 1;
    }
    else if(strcmp(argv[0], "echo") == 0)
    {
        if(argv[1] == NULL){
            printf("\n");
            return 1;
        }
        if(*(argv[1]) == '$' && strlen(argv[1]) > 1){//避免出现不是$开头或者只输入$不输入别的字符得情况 
            char *val = argv[1]+1; // $PATH $?
            if(strcmp(val, "?") == 0)//获取最近一个进程的退出码:echo $?
            {
                printf("%d\n", lastcode);
                lastcode = 0;
            }
            else{//获取环境变量中某个变量的值：echo $myval
                const char *enval = getenv(val);
                if(enval) printf("%s\n", enval);//如果有这个环境变量，打印该环境变量的值
                else printf("\n");//没有的话直接打印一个空行
            }
            return 1;
        }
        else {
            printf("%s\n", argv[1]);
            return 1;
        }
    }
    else if(0){}

    return 0;
}

int main()
{
    while(1){
        char usercommand[NUM];
        char *argv[SIZE];
        // 1. 打印提示符&&获取用户命令字符串获取成功
        int n = getUserCommand(usercommand, sizeof(usercommand));
        if(n <= 0) continue;//如果输入的是空串（返回-1），那么不需要创建子进程并且程序替换，直接continue
        // 2. 分割字符串
        // "ls -a -l" -> "ls" "-a" "-l"
        commandSplit(usercommand, argv);//调用函数将第一个参数分割，放到第二个参数中，也就是说所有的命令放在argv里面
        // 3. 检查是不是内建命令
        n = doBuildin(argv);//需要有一个函数专门处理内建命令
        if(n) continue;
        // 4. 执行对应的命令
        execute(argv);
    }
}
